package com.asia.algorithmcode.array;

/**
 * @DESCRIPTION: 75
 * @USER: wanfu
 * @DATE: 2025/3/21 星期五 17:55
 */
public class SortColors {

    public static void main(String[] args) {
        int[] arr = new int[]{2, 0, 2, 1, 1, 0};
        new SortColors().sortColors(arr);

        for (int i = 0; i < arr.length; i++) {
            System.out.println(arr[i] + "\t");
        }
    }

    public void sortColors(int[] nums) {
        quickSort(nums, 0, nums.length - 1);
    }

    public static void quickSort(int[] arr, int low, int high) {
        if (low < high) {
            // partitionIndex是基准元素被放置的正确位置
            int partitionIndex = partition(arr, low, high);

            // 分别对基准元素左侧和右侧的子数组进行快速排序
            quickSort(arr, low, partitionIndex - 1);
            quickSort(arr, partitionIndex + 1, high);
        }
    }

    private static int partition(int[] arr, int low, int high) {
        // 选择最右边的元素作为基准
        int pivot = arr[high];
        int i = (low - 1); // i指向小于pivot的元素的最后位置

        for (int j = low; j < high; j++) {
            // 如果当前元素小于或等于pivot
            if (arr[j] <= pivot) {
                i++;

                // 交换arr[i]和arr[j]
                int temp = arr[i];
                arr[i] = arr[j];
                arr[j] = temp;
            }
        }

        // 将pivot元素放到正确的位置
        int temp = arr[i + 1];
        arr[i + 1] = arr[high];
        arr[high] = temp;

        return i + 1; // 返回pivot的正确位置
    }

}

